package com.heima.search.service.impl;

import com.heima.model.common.dtos.ResponseResult;
import com.heima.model.common.enums.AppHttpCodeEnum;
import com.heima.model.search.dtos.UserSearchDto;
import com.heima.search.pojos.ApAssociateWords;
import com.heima.search.service.ApAssociateWordsService;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.stereotype.Service;

import java.util.List;

/**
 * @Description:
 * @Version: V1.0
 */
@Service
@Slf4j
public class ApAssociateWordsServiceImpl implements ApAssociateWordsService {

    @Autowired
    MongoTemplate mongoTemplate;

    /**
     * 联想词
     * @param userSearchDto
     * @return
     */
    @Override
    //主要功能是从MongoDB中查找与用户提供的搜索词相关的联想词，并将这些联想词以分页的形式返回给调用者
    //在查找之前，方法会进行输入参数的有效性检查，并对分页大小进行限制
    public ResponseResult findAssociate(UserSearchDto userSearchDto) {
        //1 参数检查，如果对象为空或者searchWords为空，返回错误信息
        if(userSearchDto == null || StringUtils.isBlank(userSearchDto.getSearchWords())){
            //返回错误信息
            return ResponseResult.errorResult(AppHttpCodeEnum.PARAM_INVALID);
        }

        //分页检查，只显示前20条数据
        if (userSearchDto.getPageSize() > 20) {
            userSearchDto.setPageSize(20);
        }

        //3 执行查询 模糊查询，创建一个Query对象，用于构建查询条件
        //表示在associateWords字段中查找包含userSearchDto.getSearchWords()的文档
        Query query = Query.query(Criteria.where("associateWords").regex(".*?\\" + userSearchDto.getSearchWords() + ".*"));
        //使用query.limit(userSearchDto.getPageSize())来设置查询结果的分页大小，即只返回限定数量的记录
        query.limit(userSearchDto.getPageSize());
        //调用mongoTemplate.find(query, ApAssociateWords.class)方法执行查询，并将查询结果存储在wordsList变量中
        //ApAssociateWords是一个POJO类，用于映射MongoDB中的文档到Java对象
        List<ApAssociateWords> wordsList = mongoTemplate.find(query, ApAssociateWords.class);

        return ResponseResult.okResult(wordsList);
    }
}